// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/compositor/test/test_compositor_host.h"

#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "base/thread_task_runner_handle.h"
#include "ui/compositor/compositor.h"
#include "ui/gfx/win/window_impl.h"

namespace ui {

class TestCompositorHostWin : public TestCompositorHost,
                              public gfx::WindowImpl {
public:
    TestCompositorHostWin(const gfx::Rect& bounds,
        ui::ContextFactory* context_factory)
    {
        Init(NULL, bounds);
        compositor_.reset(new ui::Compositor(context_factory,
            base::ThreadTaskRunnerHandle::Get()));
        compositor_->SetAcceleratedWidget(hwnd());
        compositor_->SetScaleAndSize(1.0f, GetSize());
    }

    ~TestCompositorHostWin() override { DestroyWindow(hwnd()); }

    // Overridden from TestCompositorHost:
    void Show() override { ShowWindow(hwnd(), SW_SHOWNORMAL); }
    ui::Compositor* GetCompositor() override { return compositor_.get(); }

private:
    CR_BEGIN_MSG_MAP_EX(TestCompositorHostWin)
    CR_MSG_WM_PAINT(OnPaint)
    CR_END_MSG_MAP()

    void OnPaint(HDC dc)
    {
        compositor_->ScheduleFullRedraw();
        ValidateRect(hwnd(), NULL);
    }

    gfx::Size GetSize()
    {
        RECT r;
        GetClientRect(hwnd(), &r);
        return gfx::Rect(r).size();
    }

    scoped_ptr<ui::Compositor> compositor_;

    DISALLOW_COPY_AND_ASSIGN(TestCompositorHostWin);
};

TestCompositorHost* TestCompositorHost::Create(
    const gfx::Rect& bounds,
    ui::ContextFactory* context_factory)
{
    return new TestCompositorHostWin(bounds, context_factory);
}

} // namespace ui
